from utils import DriverUtils
from selenium.webdriver.support.wait import WebDriverWait


class BasePage(object):
    # type 就3个值  mp,mis,app
    def __init__(self, type='mp'):
        if type == 'mp':
            self.driver = DriverUtils.get_mp_driver()
        elif type == 'mis':
            self.driver = DriverUtils.get_mis_driver()
        elif type == 'app':
            self.driver = DriverUtils.get_app_driver()
        else:
            print('不好意思，没有对应的类型')

    # 查找一个元素
    def find_element(self, values):
        # 为了更好的找到我们的元素 应该对元素进行 显示等待
        return WebDriverWait(self.driver, 10, 0.5).until(lambda x: x.find_element(values[0], values[1]))

    # 查找多个元素
    def find_elements(self, values):
        # 为了更好的找到我们的元素 应该对元素进行 显示等待
        return WebDriverWait(self.driver, 10, 0.5).until(lambda x: x.find_elements(values[0], values[1]))

        # return self.driver.find_element(values[0],values[1])

# BasePage(type='pm')
# BasePage(type='mis')
# BasePage(type='app')
